from agentscope.service.service_toolkit import ServiceToolkit

from loguru import logger

class FlexServiceToolkit(ServiceToolkit):
    def Remove(self, service_func_name: str):
        if service_func_name in self.service_funcs:
            del self.service_funcs[service_func_name]
        else:
            logger.warning(f"{service_func_name} not found in service_funcs")